"""
    Script que crea una lista M3U a partir de un archivo txt con los links de una serie,
    Un link por línea.
    Recibe:
        1. Ruta y nombre del archivo txt
        2. Nombre de la serie que estará incluido en los capítulos.
        3. Número de dígitos en la numeración dependiendo del número de episodios.
            (Define el número de ceros, va de 1-4)
            Ej. Menos de 1000 episodios: 3 -> 001, 002, ...
                Menos de 100 episodios:  2 -> 01, 02, ...
 
"""

import sys

fileName = sys.argv[1]
name = sys.argv[2]
digits = sys.argv[3]
with open(fileName,'r') as f:
    links = f.readlines()
c = 0
f = open(fileName.split('.')[0]+'.m3U', 'w')
f.write("#EXTM3U\n")
for link in links:
    c = c+1
    if digits == '1':
        f.write(f"#EXTINF,0,{name}{c}\n")
    elif digits == '2':
        f.write(f"#EXTINF,0,{name}{c:02}\n")
    elif digits == '3':
        f.write(f"#EXTINF,0,{name}{c:03}\n")
    else:
        f.write(f"#EXTINF,0,{name}{c:04}\n")
    f.write(link)
f.close()

